package com.tuniu.agents.impl.train.enums;

import lombok.Getter;

import static com.tuniu.agents.impl.train.TrainAgentConfig.*;

@Getter
public enum TrainAgentEnum {

    TRAIN_TICKET_AGENT(AGENT_ID,"train-selection","tran.agent.before.message",null),
    TRAIN_PARSE_LIST_AGENT(TRAIN_PARSE_LIST_AGENT_ID,"train-recommend-all",null,null),
    TRAIN_PARSE_DETAIL_AGENT(TRAIN_PARSE_DETAIL_AGENT_ID,null,null,null),
    ;

    private String agentId;

    private String promptTemplateId;

    private String before;

    private String after;


    TrainAgentEnum(String agentId, String promptTemplateId, String before, String after){
        this.agentId = agentId;
        this.promptTemplateId = promptTemplateId;
        this.before = before;
        this.after = after;
    }


}
